import java.util.HashMap;
import java.util.Map;

/**
 * 525. 连续数组
 */
class Solution5 {
    public int findMaxLength(int[] nums) {
        Map<Integer, Integer> map = new HashMap<>();
        int n = nums.length, ret = 0, sum = 0;
        map.put(0, -1);
        for(int i = 0; i < n; i++) {
            if(nums[i] == 0) {
                nums[i] = -1;
            }   
        }

        for(int i = 0; i < n; i++) {
            sum += nums[i];
            if(map.containsKey(sum)) {
                ret = Math.max(ret, i - map.get(sum));
            }
            map.put(sum, Math.min(i, map.getOrDefault(sum, Integer.MAX_VALUE)));
        }

        return ret;
    }
}